package cn.t.repository.impl.user;

import cn.t.converter.user.ParticipantInfoConverter;
import cn.t.core.mode.YesNoEnum;
import cn.t.core.repoistory.RepositoryImpl;
import cn.t.core.enums.base.BaseStatusEnum;
import cn.t.model.user.ParticipantInfoDO;
import cn.t.persistence.entity.user.ParticipantInfoPO;
import cn.t.persistence.mapper.user.ParticipantInfoMapper;
import cn.t.repository.user.ParticipantInfoRepository;
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Repository;
import org.springframework.util.CollectionUtils;
import org.springframework.util.ObjectUtils;
import org.springframework.util.StringUtils;

import java.time.LocalDateTime;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 参与者信息表数据仓库
 *
 * @author l
 */
@Slf4j
@Repository
@AllArgsConstructor
public class ParticipantInfoRepositoryImpl extends RepositoryImpl<ParticipantInfoMapper, ParticipantInfoPO, ParticipantInfoConverter, ParticipantInfoDO> implements ParticipantInfoRepository {


    @Override
    public List<ParticipantInfoDO> findByModel(ParticipantInfoDO participantInfoDO, int start, Integer pageSize, String field, String direction) {

        String lastSql = StringUtils.hasText(field) ? " order by " + field + " " + direction + " limit " + start + "," + pageSize : " limit " + start + "," + pageSize;

        List<ParticipantInfoPO> list = getWrapper(participantInfoDO).last(lastSql).list();
        if (CollectionUtils.isEmpty(list)) {
            return null;
        }

        return list.stream().map(converter::po2Model).collect(Collectors.toList());
    }

    private LambdaQueryChainWrapper<ParticipantInfoPO> getWrapper(ParticipantInfoDO participantInfoDO) {
        LambdaQueryChainWrapper<ParticipantInfoPO> wrapper = lambdaQuery();
        if (!ObjectUtils.isEmpty(participantInfoDO.getGmtCreatedStart()) && !ObjectUtils.isEmpty(participantInfoDO.getGmtCreatedEnd())) {
            wrapper.between(ParticipantInfoPO::getGmtCreated, participantInfoDO.getGmtCreatedStart(), participantInfoDO.getGmtCreatedEnd());
        }
        if (!ObjectUtils.isEmpty(participantInfoDO.getGmtModifiedStart()) && !ObjectUtils.isEmpty(participantInfoDO.getGmtModifiedEnd())) {
            wrapper.between(ParticipantInfoPO::getGmtModified, participantInfoDO.getGmtModifiedStart(), participantInfoDO.getGmtModifiedEnd());
        }
        if (!ObjectUtils.isEmpty(participantInfoDO.getUserId())) {
            wrapper.eq(ParticipantInfoPO::getUserId, participantInfoDO.getUserId());
        }
        if (!ObjectUtils.isEmpty(participantInfoDO.getScenarioId())) {
            wrapper.eq(ParticipantInfoPO::getScenarioId, participantInfoDO.getScenarioId());
        }
        if (!ObjectUtils.isEmpty(participantInfoDO.getIndividualId())) {
            wrapper.eq(ParticipantInfoPO::getIndividualId, participantInfoDO.getIndividualId());
        }
        if (!ObjectUtils.isEmpty(participantInfoDO.getFirmId())) {
            wrapper.eq(ParticipantInfoPO::getFirmId, participantInfoDO.getFirmId());
        }
        if (!ObjectUtils.isEmpty(participantInfoDO.getDeptId())) {
            wrapper.eq(ParticipantInfoPO::getDeptId, participantInfoDO.getDeptId());
        }
        if (!ObjectUtils.isEmpty(participantInfoDO.getParticipantIds())) {
            wrapper.in(ParticipantInfoPO::getId, participantInfoDO.getParticipantIds());
        }

        if (!ObjectUtils.isEmpty(participantInfoDO.getCostCenterId())) {
            wrapper.eq(ParticipantInfoPO::getCostCenterId, participantInfoDO.getCostCenterId());
        }

        if (!ObjectUtils.isEmpty(participantInfoDO.getStatus())) {
            wrapper.eq(ParticipantInfoPO::getStatus, participantInfoDO.getStatus());
        } else {
            wrapper.eq(ParticipantInfoPO::getStatus, BaseStatusEnum.AVAILABLE.getCode());
        }

        if (!ObjectUtils.isEmpty(participantInfoDO.getIsDeleted())) {
            wrapper.eq(ParticipantInfoPO::getIsDeleted, participantInfoDO.getIsDeleted());
        } else {
            wrapper.eq(ParticipantInfoPO::getIsDeleted, YesNoEnum.FALSE.getCode());
        }
        return wrapper;
    }

    @Override
    public long count(ParticipantInfoDO participantInfoDO) {
        return getWrapper(participantInfoDO).count();
    }

    @Override
    public boolean deleteParticipantByUserId(Long userId) {
        return lambdaUpdate().set(ParticipantInfoPO::getStatus, String.valueOf(BaseStatusEnum.DISABLE.getCode()))
                .set(ParticipantInfoPO::getIsDeleted, String.valueOf(YesNoEnum.TRUE.getCode()))
                .set(ParticipantInfoPO::getGmtModified, LocalDateTime.now()).eq(ParticipantInfoPO::getUserId, userId).update();
    }

    @Override
    public boolean disableByUserId(Long userId) {
        return lambdaUpdate().set(ParticipantInfoPO::getStatus, String.valueOf(BaseStatusEnum.DISABLE.getCode()))
                .set(ParticipantInfoPO::getGmtModified, LocalDateTime.now())
                .eq(ParticipantInfoPO::getUserId, userId)
                .eq(ParticipantInfoPO::getStatus, String.valueOf(BaseStatusEnum.AVAILABLE.getCode()))
                .eq(ParticipantInfoPO::getIsDeleted, String.valueOf(YesNoEnum.FALSE.getCode())).update();
    }

    @Override
    public void clearCostCenter(Long userId) {
        lambdaUpdate().set(ParticipantInfoPO::getCostCenterId, null)
                .eq(ParticipantInfoPO::getUserId, userId)
                .eq(ParticipantInfoPO::getStatus, String.valueOf(BaseStatusEnum.AVAILABLE.getCode()))
                .eq(ParticipantInfoPO::getIsDeleted, String.valueOf(YesNoEnum.FALSE.getCode())).update();
    }

    @Override
    public void disableParticipantByUserId(Long userId) {
        lambdaUpdate().set(ParticipantInfoPO::getStatus, String.valueOf(BaseStatusEnum.DISABLE.getCode()))
                .set(ParticipantInfoPO::getIsDeleted, String.valueOf(YesNoEnum.FALSE.getCode()))
                .eq(ParticipantInfoPO::getUserId, userId).update();
    }

    @Override
    public void enableParticipantByUserId(Long userId) {
        lambdaUpdate().set(ParticipantInfoPO::getStatus, String.valueOf(BaseStatusEnum.AVAILABLE.getCode()))
                .set(ParticipantInfoPO::getIsDeleted, String.valueOf(YesNoEnum.FALSE.getCode()))
                .eq(ParticipantInfoPO::getUserId, userId).update();
    }

    @Override
    public void unbindCostCenter(Long id) {
        lambdaUpdate().set(ParticipantInfoPO::getCostCenterId, null)
                .eq(ParticipantInfoPO::getId, id).update();
    }

}
